package com.itheima.reggie.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.itheima.reggie.common.R;
import com.itheima.reggie.entity.User;
import com.itheima.reggie.service.UserService;
import com.itheima.reggie.utils.SMSUtils;
import com.itheima.reggie.utils.ValidateCodeUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@RestController
@RequestMapping("/user")
@Slf4j
/* 优化验证码5分钟过期
    1.生成验证码之后，以前存在session中，优化后存储到redis（设置过期时间5分钟）
    2.登录的时候，以前从session获取真实验证码来比较，优化后从redis中获取真实的验证码
    3.保证验证码只用一次，取出验证码之后立刻就要销毁redis中这个真实的验证码。
 */
public class UserController {
    @Autowired
    private UserService userService;

    @Autowired
    private RedisTemplate redisTemplate;
    /*
        为用户的手机号发送一条随机的验证码短信，并把验证码存储在session中，key是手机号
     */
    // {"phone":13688899998}
    @PostMapping("sendMsg")
    public R sendMsg(@RequestBody Map<String,String> map, HttpServletRequest req){
        //获取手机号
        String phoneNum = map.get("phone");
        //生成验证码
        String code = ValidateCodeUtils.generateValidateCode4String(6);
        //发送短信
        //SMSUtils.sendMessage(phoneNum,code);
        log.info("您的验证码为:{}",code);
        //把验证码存储在session中，方便登录的时候去比较
        //req.getSession().setAttribute("code_"+phoneNum,code);
        //把验证码存入redis
        redisTemplate.opsForValue().set("code_"+phoneNum,code,5, TimeUnit.MINUTES);
        return R.success(null);
    }

    @PostMapping("/login")
    public R login(@RequestBody Map<String,String> map, HttpServletRequest req){
        //获取手机号和用户填写的验证码
        String phoneNum = map.get("phone");
        String code = map.get("code");
        //根据手机号从session中获取真实的验证码
//        String codeForSession = (String) req.getSession().getAttribute("code_" + phoneNum);
        //从redis中获取验证码
        String codeForRedis = (String) redisTemplate.opsForValue().get("code_" + phoneNum);

        //比较验证码是否相同
        if(code!=null && code.equals(codeForRedis)){
            //判断当前登录的用户是否为新用户，如果是新用户就保存用户数据到数据库中。
            LambdaQueryWrapper<User> userLambdaQueryWrapper = new LambdaQueryWrapper<>();
            userLambdaQueryWrapper.eq(User::getPhone,phoneNum);
            User user = userService.getOne(userLambdaQueryWrapper);
            if(user==null){
                user = new User();
                user.setPhone(phoneNum);
                user.setStatus(1);
                userService.save(user);
            }
            //登录成功,记录登录成功的状态，把user存储在session中
            req.getSession().setAttribute("user",user);
            //登录成功之后，删除验证码
            redisTemplate.delete("code_" + phoneNum);//删除验证码
            return R.success(null);
        }

        return R.error("验证码有误");
    }

    @PostMapping("/loginout")
    public R logout(HttpServletRequest req){
        req.getSession().invalidate();
        return R.success(null);
    }
}
